#include "../mqserver/mq_exchange.hpp"
#include <gtest/gtest.h>

// 全局交换器管理器实例指针，用于测试
zdfmq::ExchangeManager::ptr emp;

/**
 * 测试环境初始化类
 * 负责在测试套件执行前后进行全局资源管理
 */
class ExchangeTest : public testing::Environment
{
public:
    /**
     * 测试套件启动前执行
     * - 初始化交换器管理器
     * - 指定元数据存储路径为./data/meta.db
     */
    virtual void SetUp() override
    {
        emp = std::make_shared<zdfmq::ExchangeManager>("./data/meta.db");
    }

    /**
     * 测试套件结束后执行
     * - 可添加资源清理逻辑
     * - 当前仅打印清理提示
     */
    virtual void TearDown() override
    {
        // emp->clear(); // 取消注释可在测试后清空数据
        std::cout << "最后的清理！！\n";
    }
};

/**
 * 交换器插入功能测试
 * - 测试连续声明4个交换器
 * - 验证交换器总数是否正确
 */
TEST(exchange_test, insert_test)
{
    const google::protobuf::Map<std::string, std::string> map;

    // 声明4个持久化直连交换器
    emp->declareExchange("exchange1", zdfmq::ExchangeType::DIRECT, true, false, map);
    emp->declareExchange("exchange2", zdfmq::ExchangeType::DIRECT, true, false, map);
    emp->declareExchange("exchange3", zdfmq::ExchangeType::DIRECT, true, false, map);
    emp->declareExchange("exchange4", zdfmq::ExchangeType::DIRECT, true, false, map);

    // 验证交换器总数
    ASSERT_EQ(emp->size(), 4);
}

/**
 * 交换器查询功能测试
 * - 验证已存在交换器的查询
 * - 验证不存在交换器的查询
 * - 验证交换器属性的正确性
 */
TEST(exchange_test, select_test)
{
    // 验证交换器存在性
    ASSERT_EQ(emp->exists("exchange1"), true);
    ASSERT_EQ(emp->exists("exchange2"), true);
    ASSERT_EQ(emp->exists("exchange3"), true);
    ASSERT_EQ(emp->exists("exchange4"), true);

    // 查询交换器并验证属性
    zdfmq::Exchange::ptr exp = emp->selectExchange("exchange2");
    ASSERT_NE(exp.get(), nullptr);
    ASSERT_EQ(exp->name, "exchange2");
    ASSERT_EQ(exp->durable, true);
    ASSERT_EQ(exp->auto_delete, false);
    ASSERT_EQ(exp->type, zdfmq::ExchangeType::DIRECT);
    exp->setArgs(std::string("k1=v1&k2=v2&"));
    ASSERT_EQ(exp->getArgs(), std::string("k1=v1&k2=v2&"));
}

/**
 * 交换器删除功能测试
 * - 验证删除后交换器是否不存在
 */
TEST(exchange_test, remove_test)
{
    // 删除交换器
    emp->deleteExchange("exchange2");

    // 验证交换器已不存在
    zdfmq::Exchange::ptr exp = emp->selectExchange("exchange2");
    ASSERT_EQ(exp.get(), nullptr);
    ASSERT_EQ(emp->exists("exchange2"), false);
}

/**
 * 测试入口函数
 * - 初始化Google Test框架
 * - 注册全局测试环境
 * - 执行所有测试用例
 */
int main(int argc, char *argv[])
{
    testing::InitGoogleTest(&argc, argv);
    testing::AddGlobalTestEnvironment(new ExchangeTest);
    return RUN_ALL_TESTS();
}